package org.mx.spring.utils;

import java.util.Map;

import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

/**
 * 全局访问Spring IoC中的Bean的工具类
 *
 * @author : john.peng created on date : 2018/1/5
 */
@Component("springContextHolder")
public class SpringContextHolder implements ApplicationContextAware {
	private static ApplicationContext applicationContext;

	/**
	 * 取得存储在静态变量中的ApplicationContext.
	 *
	 * @return 上下文
	 */
	public static ApplicationContext getApplicationContext() {
		checkApplicationContext();
		return applicationContext;
	}

	/**
	 * 实现ApplicationContextAware接口的context注入函数, 将其存入静态变量.
	 *
	 * @param applicationContext 上下文
	 */
	@Override
	public void setApplicationContext(ApplicationContext applicationContext) {
		SpringContextHolder.applicationContext = applicationContext;
	}

	/**
	 * 从静态变量ApplicationContext中取得Bean, 自动转型为所赋值对象的类型.
	 *
	 * @param name  Bean的名称
	 * @param clazz 类
	 * @param <T>   泛型类型
	 * @return Bean
	 */
	public static <T> T getBean(String name, Class<T> clazz) {
		checkApplicationContext();
		return applicationContext.getBean(name, clazz);
	}

	/**
	 * 判定指定名称的Bean是否存在于IoC容器中
	 * @param name Bean名称
	 * @return 如果存在返回true
	 */
	public static boolean contains(String name) {
		checkApplicationContext();
		return applicationContext.containsBean(name);
	}

	/**
	 * 从静态变量ApplicationContext中取得Bean, 自动转型为所赋值对象的类型.
	 * 如果有多个Bean符合Class, 取出第一个.
	 *
	 * @param clazz 类
	 * @param <T>   泛型类型
	 * @return Bean
	 */
	public static <T> T getBean(Class<T> clazz) {
		checkApplicationContext();
		Map<String, T> beanMaps = applicationContext.getBeansOfType(clazz);
		if (beanMaps != null && !beanMaps.isEmpty()) {
			return (T) beanMaps.values().iterator().next();
		} else {
			return null;
		}
	}

	/**
	 * 判定上下文是否设置
	 */
	private static void checkApplicationContext() {
		if (applicationContext == null) {
			throw new IllegalStateException("The application context is null.");
		}
	}
}
